package com.ruoyi.common.enums

import java.util.function.Function
import com.ruoyi.common.utils.DesensitizedUtil

/**
 * 脱敏类型
 *
 * @author ruoyi
 */
enum class DesensitizedType(desensitizer: Function<String, String>) {
    /**
     * 姓名，第2位星号替换
     */
    USERNAME(Function<String, String> { s -> s.replace(Regex("(\\S)\\S(\\S*)"), "$1*$2") }),

    /**
     * 密码，全部字符都用*代替
     */
    PASSWORD(DesensitizedUtil::password),

    /**
     * 身份证，中间10位星号替换
     */
    ID_CARD(Function<String, String> { s -> s.replace(Regex("(\\d{4})\\d{10}(\\d{4})"), "$1** **** ****$2") }),

    /**
     * 手机号，中间4位星号替换
     */
    PHONE(Function<String, String> { s -> s.replace(Regex("(\\d{3})\\d{4}(\\d{4})"), "$1****$2") }),

    /**
     * 电子邮箱，仅显示第一个字母和@后面的地址显示，其他星号替换
     */
    EMAIL(Function<String, String> { s -> s.replace(Regex("(^.)[^@]*(@.*$)"), "$1****$2") }),

    /**
     * 银行卡号，保留最后4位，其他星号替换
     */
    BANK_CARD(Function<String, String> { s -> s.replace(Regex("\\d{15}(\\d{3})"), "**** **** **** **** $1") }),

    /**
     * 车牌号码，包含普通车辆、新能源车辆
     */
    CAR_LICENSE(DesensitizedUtil::carLicense);

    private val desensitizer: Function<String, String>

    init {
        this.desensitizer = desensitizer
    }

    fun desensitizer(): Function<String, String> {
        return desensitizer
    }
}
